package com.mango.handler;

import com.mango.response.Result;
import com.mango.response.ResultCode;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;

/**
 * @Author:  MangoCookieMilk
 * @Date:     2021/1/13 21:59
 * 统一异常处理类  使用 @ControllerAdvice 实现全局异常处理，只需要定义类，
 */
@ControllerAdvice
@Slf4j
public class GlobalExceptionHandler {

    private static final Logger log = LoggerFactory.getLogger(GlobalExceptionHandler.class);
    /**
     * @ExceptionHandler 注解用来指明异常的处理类型，
     * 即如果这里指定为 Exception，则所有的异常都会到这里来。
     * @return
     * @ResponseBody  这里用这个注解是因为在controller中使用的是RestController
     */
    @ExceptionHandler(Exception.class)
    @ResponseBody
    public Result error (Exception e){
        e.printStackTrace();
        return Result.error();
    }

    @ExceptionHandler(ArithmeticException.class)
    @ResponseBody
    public Result error (ArithmeticException e){
        log.error(e.getMessage());
        return Result.error().code(ResultCode.ARITHMETIC_EXCEPTION.getCode())
                .message(ResultCode.ARITHMETIC_EXCEPTION.getMessage());
    }

    @ExceptionHandler(BusinessException.class)
    @ResponseBody
    public Result error (BusinessException e){
        log.error(e.getErrMsg());
        return Result.error().code(e.getCode())
                .message(e.getErrMsg());
    }
}
